import pandas as pd

def get_train_data(x_path = '',y_path = ''):
    # set data path
    train_x_csv = x_path
    train_y_csv = y_path
    # load data
    train_x = pd.read_csv(train_x_csv)
    train_y = pd.read_csv(train_y_csv)
    train_xy = pd.merge(train_x, train_y, on='uid')
    return train_xy

def get_test_data(file_path = ''):
    test = pd.read_csv(file_path)
    test_uid = test.uid
    test_x = test.drop(['uid'], axis = 1)
    return test_uid,test_x